import torch

# 加载 YOLOv8 small 模型
model = torch.hub.load('ultralytics/yolov8', 'yolov8s')

# 或者使用具体的权重文件
# model = torch.hub.load('ultralytics/yolov8', 'custom', 'path/to/yolov8_s.pth')
